set(builder_src
    ${src_dir}/trt_builder/trt_builder.cpp
)

set(builder_headers
    ${src_dir}/trt_builder/trt_builder.h
)

add_library(trt_builder SHARED ${builder_src} ${builder_headers})
target_include_directories(trt_builder PRIVATE ${src_dir})
target_link_libraries(trt_builder PRIVATE trt_utils)
link_cuda_trt(trt_builder)
add_export_define(trt_builder)
